CUDA_VISIBLE_DEVICES=1 python -u train_new.py